package com.leetcode.explore;

import java.util.HashMap;
import java.util.Map;

/*

两数之和

给定一个整数数组 nums和一个目标值 target，请你在该数组中找出和为目标值的那两个整数，并返回他们的数组下标。
你可以假设每种输入只会对应一个答案。但是，数组中同一个元素不能使用两遍。

示例:
给定 nums = [2, 7, 11, 15], target = 9
因为 nums[0] + nums[1] = 2 + 7 = 9
所以返回 [0, 1]

*/

public class Prob0001 {
	class Solution {
		public int[] twoSum(int[] nums, int target) {
			Map<Integer, Integer> map = new HashMap<>();
			for (int i = 0; i < nums.length; ++i) {
				int temp = target - nums[i];
				if (map.containsKey(temp)) {
					return new int[] { map.get(temp), i };
				}
				map.put(nums[i], i);
			}
			throw new IllegalArgumentException("No two sum solution!");
		}
	}
}
